package dao

import (
	"database/sql"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var DB *gorm.DB
var sqlDB *sql.DB

func InitDatabase() (err error) {
	dsn := "root:HYJ=wvs1996@tcp(127.0.0.1:3306)/ToDo?charset=utf8mb4&parseTime=true&loc=Local"
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return
	}
	//获取获取通用数据库对象 *sql.DB
	sqlDB, err = DB.DB()
	if err != nil {
		return
	}

	//验证数据库是否建立连接,出问题返回error对象，没出问题返回nil
	return sqlDB.Ping()

}

func CloseLinkedDatabase() {
	sqlDB.Close()
}
